package com.ruoyi.aisite.controller;

import com.ruoyi.aisite.domain.AiTemplate;
import com.ruoyi.aisite.service.IAiTemplateService;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.core.text.Convert;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * AI模板Controller
 * 
 * @author ruoyi
 */
@Controller
@RequestMapping("/aisite/template")
public class AiTemplateController extends BaseController {
    private String prefix = "aisite/template";

    @Autowired
    private IAiTemplateService aiTemplateService;

    @GetMapping()
    public String template() {
        return prefix + "/template";
    }

    /**
     * 查询AI模板列表
     */
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(AiTemplate aiTemplate) {
        startPage();
        List<AiTemplate> list = aiTemplateService.selectAiTemplateList(aiTemplate);
        return getDataTable(list);
    }

    /**
     * 新增AI模板
     */
    @GetMapping("/add")
    public String add() {
        return prefix + "/add";
    }

    /**
     * 新增保存AI模板
     */
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(AiTemplate aiTemplate) {
        return toAjax(aiTemplateService.insertAiTemplate(aiTemplate));
    }

    /**
     * 修改AI模板
     */
    @GetMapping("/edit/{templateId}")
    public String edit(@PathVariable("templateId") Long templateId, ModelMap mmap) {
        AiTemplate aiTemplate = aiTemplateService.selectAiTemplateById(templateId);
        mmap.put("aiTemplate", aiTemplate);
        return prefix + "/edit";
    }

    /**
     * 修改保存AI模板
     */
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(AiTemplate aiTemplate) {
        return toAjax(aiTemplateService.updateAiTemplate(aiTemplate));
    }

    /**
     * 删除AI模板
     */
    @PostMapping("/remove")
    @ResponseBody
    public AjaxResult remove(String ids) {
        return toAjax(aiTemplateService.deleteAiTemplateByIds(Convert.toLongArray(ids)));
    }
}